package com.heima.wemedia.service.impl;

import com.alibaba.fastjson.JSONArray;
import com.heima.aliyun.scan.GreenScan;
import com.heima.aliyun.scan.ScanResult;
import com.heima.common.exception.CustException;
import com.heima.feigns.AdminFeign;
import com.heima.model.common.constants.message.PublishArticleConstants;
import com.heima.model.common.constants.wemedia.WemediaConstants;
import com.heima.model.common.dtos.ResponseResult;
import com.heima.model.common.enums.AppHttpCodeEnum;
import com.heima.model.wemedia.pojos.WmNews;
import com.heima.utils.common.SensitiveWordUtil;
import com.heima.wemedia.mapper.WmNewsMapper;
import com.heima.wemedia.service.WmNewsAutoScanService;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang3.StringUtils;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.util.CollectionUtils;

import java.time.LocalDateTime;
import java.util.*;
import java.util.stream.Collectors;
@Service
@Slf4j
public class WmNewsAutoScanServiceImpl implements WmNewsAutoScanService {
    @Autowired
    private WmNewsMapper wmNewsMapper;
    @Autowired
    private RabbitTemplate rabbitTemplate;
    @Value("${file.oss.web-site}")
    String webSite;
    /**
     * 自动审核方法
     * @param wmNewsId
     */
    @Override
    public void autoScanWmNews(Integer wmNewsId) {
        log.info(" 自动审核发布方法 被调用   当前审核发布的文章id==> {}",wmNewsId);
        //1. 根据文章id 远程调用feign查询文章
        if (wmNewsId == null) {
            log.error("自动审核文章失败    文章id为空");
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID,"待审核的文章id为空");
        }
        WmNews wmNews = wmNewsMapper.selectById(wmNewsId);
        if (wmNews==null) {
            log.error("自动审核文章失败    未查询自媒体文章信息  wmNewsId:{}",wmNewsId);
            CustException.cust(AppHttpCodeEnum.DATA_NOT_EXIST,"对应的文章不存在");
        }
        // 2. 判断文章状态是否为待审核状态  必须是 1 (待审核)  (避免重复消费)
        Short status = wmNews.getStatus();
        if(!WemediaConstants.WM_NEWS_SUMMIT_STATUS.equals(status)){
            log.info("当前文章状态为: {}  不是待审核状态,无需审核",status);
            return;
        }
        // 抽取出文章中 所有的文本内容 和 所有的图片url集合    Map<String,Object>  content 内容   images List<String>
        Map<String,Object> contentAndImageResult = handleTextAndImages(wmNews);
        // 3. DFA 进行自管理敏感词审核  2 有敏感词不通过  通过继续下一步
        boolean scanSensitive = handleSensitive((String)contentAndImageResult.get("content"),wmNews);
        if(!scanSensitive){
            // 返回false代表审核不成功  结束处理
            log.info("文章审核未通过，原因：内容中包含敏感词");
            return;
        }
        log.info(" 自管理敏感词审核通过  =======   ");

        // 4.  阿里云的文本审核   失败  状态2  不确定 状态3
        boolean ScanText = handleTextScan((String)contentAndImageResult.get("content"),wmNews);
        if(!ScanText){
            // 返回false代表审核不成功  结束处理
            log.info("文章审核未通过，原因：内容中包含违规词汇");
            return;
        }
        log.info(" 阿里云内容审核通过  =======   ");


        // 5.  阿里云的图片审核  2 有违规图片  3 不确定/aliyun未调用成功  通过继续下一步
        boolean ScanImages =handleImageScan((List<String>)contentAndImageResult.get("images"),wmNews);
        if(!ScanImages){
            // 返回false代表审核不成功  结束处理
            log.info("文章审核未通过，原因：图片中包含违规信息");
            return;
        }
        // 6. 将文章状态改为 8  自动审核通过
        updateWmNews(WmNews.Status.SUCCESS.getCode(),"审核通过",wmNews);

        // 7. 根据文章的发布时间，发送延迟消息 用于定时发布文章( 9 已发布)
        // 获取发布时间
        long publishTime = wmNews.getPublishTime().getTime();
        // 获取当前时间
        long nowTime = new Date().getTime();
        // 发布时间 - 当前时间 = 距离发布的延迟时间
        long remainTime = publishTime - nowTime;
        // 使用rabbittemplate 发送延迟消息
        rabbitTemplate.convertAndSend(PublishArticleConstants.DELAY_DIRECT_EXCHANGE,
                PublishArticleConstants.PUBLISH_ARTICLE_ROUTE_KEY,
                wmNews.getId(),
                message -> {                              // 延时消息 必设置
                    message.getMessageProperties().setHeader("x-delay",remainTime <= 0 ? 0 : remainTime);
                    return message;
                }
        );
        log.info("成功发送  文章延迟发布消息  文章id : {}  当前时间: {}", wmNews.getId(), LocalDateTime.now());
    }
    /**
     * 阿里云图片审核
     * @param images  待审核的图片列表
     * @return
     */
    private boolean handleImageScan(List<String> images,WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.imageUrlScan(images);
            switch (scanResult.getSuggestion()){
                case "block":
                    // 失败
                    updateWmNews(WmNews.Status.FAIL.getCode(),"文章图片中包含违规信息："+scanResult.getLabel(),wmNews);
                    break;
                case "review":
                    // 人工审核
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(),"文章图片中包含不确定内容，需要进一步人工审核",wmNews);
                    break;
                case "pass":
                    // 通过
                    flag = true;
                    break;
                default:
                    // 人工审核
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(),"阿里云调用状态异常，需要进一步人工审核",wmNews);
                    break;
            }
        } catch (Exception e) {
            e.printStackTrace();
            // 人工审核
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(),"阿里云调用失败，需要进一步人工审核",wmNews);
        }
        return flag;
    }
    @Autowired
    GreenScan greenScan;
    /**
     * 阿里云文本
     * @param content    block: 状态2    review: 状态3    异常: 状态3
     * @param wmNews
     * @return
     */
    private boolean handleTextScan(String content, WmNews wmNews) {
        boolean flag = false;
        try {
            ScanResult scanResult = greenScan.greenTextScan(content);
            // 阿里云建议： block: 违规    review: 进一步人工审核    pass: 通过
            String suggestion = scanResult.getSuggestion();
            switch (suggestion){
                case "block":
                    // 失败
                    updateWmNews(WmNews.Status.FAIL.getCode(),"文章内容中包含违规词汇："+scanResult.getLabel(),wmNews);
                    break;
                case "review":
                    // 人工审核
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(),"文章内容中包含不确定内容，需要进一步人工审核",wmNews);
                    break;
                case "pass":
                    // 通过
                    flag = true;
                    break;
                default:
                    // 人工审核
                    updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(),"阿里云调用状态异常，需要进一步人工审核",wmNews);
                    break;
            }

        } catch (Exception e) {
            e.printStackTrace();
            // 人工审核
            updateWmNews(WmNews.Status.ADMIN_AUTH.getCode(),"阿里云调用失败，需要进一步人工审核",wmNews);
        }
        return flag;
    }


    @Autowired
    AdminFeign adminFeign;

    /**
     * 基于DFA 检测内容是否包含敏感词
     * @param content 文章的所有文本内容
     * @param wmNews 文章对象
     * @return
     */
    private boolean handleSensitive(String content, WmNews wmNews) {
        boolean flag = true;
        // 1. 查询出数据库中的敏感词
        ResponseResult<List<String>> sensitives = adminFeign.sensitives();
        if(!sensitives.checkCode()){
            CustException.cust(AppHttpCodeEnum.REMOTE_SERVER_ERROR,sensitives.getErrorMessage());
        }
        List<String> sensitivesData = sensitives.getData();
        // 2. 将敏感词集合转发DFA数据模型
        SensitiveWordUtil.initMap(sensitivesData);
        // 3. 检测敏感词
        Map<String, Integer> resultMap = SensitiveWordUtil.matchWords(content);
        // 4. 如果有敏感词，将文章状态修改为 2，并告知原因
        if(!CollectionUtils.isEmpty(resultMap)){
            // 将文章状态改为2
            updateWmNews(WmNews.Status.FAIL.getCode(),"内容中包含敏感词: " + resultMap, wmNews);
            flag = false;
        }
        return flag;
    }


    /**
     * 修改文章状态
     * @param wmNews
     * @param status
     * @param reason
     */
    private void updateWmNews(short status, String reason, WmNews wmNews) {
        wmNews.setStatus(status);
        wmNews.setReason(reason);
        wmNewsMapper.updateById(wmNews);
    }

    /**
     * 抽取 文章中所有 文本内容  及 所有图片路径
     * @param wmNews  文章
     * @return   Map<String,Object>  key:  content  value:  文本内容 String   key: images  value: 图片集合  List<String>
     */
    private Map<String, Object> handleTextAndImages(WmNews wmNews) {
        String contentJson = wmNews.getContent();
        // 1. 判断 内容 不能为空  并转为 List<Map>  [{},{},{}]
        if (StringUtils.isBlank(contentJson)) {
            log.error("自动审核文章失败    文章内容为空");
            CustException.cust(AppHttpCodeEnum.PARAM_INVALID,"文章内容为空");
        }
        List<Map> contentMaps = JSONArray.parseArray(contentJson, Map.class);
        // 2. 抽取文章中所有的文本内容  并拼接成一个字符串
        //   [{type: text, value: 文本},{type: image, value: url},{}]
        //    家乡很美_hmtt_国家强大
        String content = contentMaps.stream()
                .filter(map -> "text".equals(map.get("type")))
                .map(map -> map.get("value").toString())
                .collect(Collectors.joining("_hmtt_"));
        // 将文本内容 和 标题拼接成一个字符串  多个建议使用 StringBuilder StringBuffer
        content = content + "_hmtt_" + wmNews.getTitle();
        // 3. 抽取文章中所有图片   content :  全路径       images :  文件名称  + 访问前缀
        List<String> imageList = contentMaps.stream()
                .filter(map -> "image".equals(map.get("type")))
                .map(map -> (String) map.get("value"))
                .collect(Collectors.toList());
        if (StringUtils.isNotBlank(wmNews.getImages())) {
            // 按照 逗号 切割封面字符串  得到数组   基于数组得到stream   将每一条数据都拼接一个前缀 收集成集合
            List<String> coverImages = Arrays.stream(wmNews.getImages().split(","))
                    .map(url -> webSite + url)
                    .collect(Collectors.toList());
            // 合并 内容图片 和 封面图片
            imageList.addAll(coverImages);
        }
        // 去重
        imageList = imageList.stream().distinct().collect(Collectors.toList());
        // 4. 将抽取出来的数据封装成一个map  并返回结果
        Map result = new HashMap();
        result.put("content",content);
        result.put("images",imageList);
        return result;
    }
}
